Count List Function in Python